package com.swotxu.mall.auth.controller;

import com.google.common.collect.Maps;
import com.swotxu.common.constant.AuthServerConstant;
import com.swotxu.common.constant.MallUrlConstant;
import com.swotxu.common.exception.BizCodeEnum;
import com.swotxu.common.utils.R;
import com.swotxu.common.vo.MemberRespVo;
import com.swotxu.mall.auth.feign.MemberFeignService;
import com.swotxu.mall.auth.feign.ThirdPartFeignService;
import com.swotxu.mall.auth.vo.UserLoginVo;
import com.swotxu.mall.auth.vo.UserRegistVo;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.validation.BindingResult;
import org.springframework.validation.FieldError;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.servlet.mvc.support.RedirectAttributes;

import javax.servlet.http.HttpSession;
import javax.validation.Valid;
import java.util.Map;
import java.util.UUID;
import java.util.concurrent.TimeUnit;
import java.util.stream.Collectors;

/**
 * @Date: 2021/2/15 16:57
 * @Author: swotXu
 */
@Slf4j
@Controller
public class LoginController {

    @Autowired
    StringRedisTemplate stringRedisTemplate;
    @Autowired
    ThirdPartFeignService thirdPartFeignService;
    @Autowired
    MemberFeignService memberFeignService;

    /**
     * Auth服务登录页
     *
     * @return
     */
    @GetMapping("/login.html")
    public String loginPage(HttpSession session) {
        Object attribute = session.getAttribute(AuthServerConstant.SESSION_LOGIN_USER);

        return attribute == null? "login" : MallUrlConstant.REDIRECT_MALL_MAIN_INDEX;
    }

    /**
     * 1、接口防刷
     * 2、验证码再校验
     *
     * @param phone
     * @return
     */
    @ResponseBody
    @GetMapping("/sms/sendCode")
    public R sendCode(@RequestParam("phone") String phone) {
        // 缓存验证码, 防止同一个phone在60s内多次发送
        String redisCodeKey = AuthServerConstant.keyForSmsCodeCache(phone);
        String redisCode = stringRedisTemplate.opsForValue().get(redisCodeKey);
        if (StringUtils.isNotBlank(redisCode)){
            long old = Long.parseLong(redisCode.split("_")[1]);
            if (System.currentTimeMillis() - old < 60000) {
                log.info("{}", BizCodeEnum.SMS_CODE_EXCEPTION.getMessage());
                // 60s内不能再发 - 频率校验
                return R.error(BizCodeEnum.SMS_CODE_EXCEPTION);
            }
        }
        // TODO 接口防刷
        String code = UUID.randomUUID().toString().substring(0, 4);
        stringRedisTemplate.opsForValue().set(redisCodeKey, code + "_"+System.currentTimeMillis(), 10, TimeUnit.MINUTES);
        // 第三方服务发送验证码
        thirdPartFeignService.sendCode(phone, code);
        return R.ok();
    }

    /**
     * 注册操作，成功后回到首页或登录页
     *
     * @param userRegistVo
     * @param result
     * @param model     用于直接返回数据给前端，可用于转发
     * @param attributes    用于重定向请求时，携带数据。模拟 Session，将数据存入回话中，重定向后，从会话中获取信息并删除。
     *                      //TODO 需解决分布式下的 Session问题
     * @return
     */
    @PostMapping("/regist")
    public String regist(@Valid UserRegistVo userRegistVo, BindingResult result, Model model, RedirectAttributes attributes) {
        /**
         * 使用 return "forward:/reg.html"; 转发方式。
         * 报错如下：
         *      405 Request method 'POST' not supported.
         *
         * 原因：
         * from表单提交 POST 请求，代码处理后，通过 "forward:/reg.html" 方式进行转发，
         * 而这种方式只支持 GET请求方式。
         *
         * 所以应改为下面这种方式
         * return "reg";
         *
         * 最后，由于注册请求为 POST请求，若采用转发形式，刷新页面会导致请求重复提交，应改为重定向
         */
        Map<String, String> errors;
        // 1、请求参数校验出错，转发到注册页
        if (result.hasErrors()) {
            log.warn("注册失败！原因: 提交数据有误");
            errors = result.getFieldErrors().stream()
                    .collect(Collectors.toMap(FieldError::getField, FieldError::getDefaultMessage));
            // model.addAttribute("errors", errors);
            attributes.addFlashAttribute("errors", errors);
            return MallUrlConstant.AUTH_REDIRECT_REG;
        }

        String redisCodeKey = AuthServerConstant.keyForSmsCodeCache(userRegistVo.getPhone());
        // 2、校验验证码
        String code = userRegistVo.getCode();
        String redisCode = stringRedisTemplate.opsForValue().get(redisCodeKey);
        if (StringUtils.isBlank(redisCode) || !code.equalsIgnoreCase(redisCode.split("_")[0])){
            log.warn("注册失败！原因: 验证码错误");
            attributes.addFlashAttribute("errors", newErrorMap("code", "验证码错误"));
            return MallUrlConstant.AUTH_REDIRECT_REG;
        }
        // 3、校验成功后，删除验证码
        stringRedisTemplate.delete(redisCodeKey);
        // 4、调用远程服务，开始注册
        R registResult = memberFeignService.regist(userRegistVo);
        if (!registResult.isOk()) {
            log.warn("注册失败！原因: {}", registResult.getMessage());
            attributes.addFlashAttribute("errors", newErrorMap("msg", registResult.getMessage()));
            return MallUrlConstant.AUTH_REDIRECT_REG;
        }
        // 5、成功后重定向到登陆页
        return MallUrlConstant.AUTH_REDIRECT_LOGIN;
    }

    private Map<String, String> newErrorMap(String key, String msg) {
        Map<String, String> errors = Maps.newHashMap();
        errors.put(key, msg);
        return errors;
    }

    /**
     * 登陆，成功后跳到商城首页
     * @param userLoginVo
     * @return
     */
    @PostMapping("/login")
    public String login(UserLoginVo userLoginVo, RedirectAttributes attributes, HttpSession session) {
        // 调用远程服务进行用户登录认证
        R loginResult = memberFeignService.login(userLoginVo);
        if (loginResult.isOk()) {
            MemberRespVo respVo = loginResult.getData(MemberRespVo.class);
            log.info("登录成功！{}", respVo);
            session.setAttribute(AuthServerConstant.SESSION_LOGIN_USER, respVo);
            return MallUrlConstant.REDIRECT_MALL_MAIN_INDEX;
        }
        // 登陆失败
        attributes.addFlashAttribute("errors", newErrorMap("msg", loginResult.getMessage()));
        return MallUrlConstant.AUTH_REDIRECT_LOGIN;
    }
}
